#include <brpc/server.h>
#include <butil/logging.h>
#include "main.pb.h"

// 自定义服务类：继承 Protobuf 生成的 EchoService 基类
class EchoServiceImpl : public example::EchoService {
public:
    void Echo(google::protobuf::RpcController* controller,
              const ::example::EchoRequest* request, //客户端请求
              ::example::EchoResponse* response,     //服务端请求
              ::google::protobuf::Closure* done) {   //回调（用于通知RPC完成）
                
            //自动管理done的生命周期（避免内存泄漏）
            brpc::ClosureGuard rpc_guard(done);  

            //处理请求：打印收到的消息
            std::cout << "收到消息：" << request->message() << std::endl;

            //构造响应：在请求消息后添加后缀
            std::string str = request->message() + "--这是响应！！";
            response->set_message(str);
        }
};

int main(int argc, char* agrv[]) {
    //服务改造思想: 
    //1. 关闭BRPC默认的日志（可选）
    logging::LoggingSettings settings;
    settings.logging_dest = logging::LoggingDestination::LOG_TO_NONE;
    logging::InitLogging(settings);

    //2. 初始化BRPC服务器
    brpc::Server server;

    //3. 注册服务：将EchoServiceImpl实例添加到服务器
    EchoServiceImpl echo_service;
    int ret = server.AddService(&echo_service,
    brpc::ServiceOwnership::SERVER_DOESNT_OWN_SERVICE);

    if (ret == -1) {
        std::cout << "添加Rpc服务失败!\n";
        return -1;
    }

    //4. 配置并启动服务器
    brpc::ServerOptions option;
    option.idle_timeout_sec = -1; //链接无超时（-1表示禁用）
    option.num_threads = 1;       //IO线程数量为1
    ret = server.Start(8080, &option); //监听8080端口
    if (ret == -1) {
        std::cout << "启动服务器失败！\n";
        return -1;
    }

    //阻塞等待服务器退出（直到收到终止信号）
    server.RunUntilAskedToQuit();
    return 0;
}